import taichi as ti

ti.init(arch=ti.cuda)

#传参需要指定类型 
# #但目前不支持直接传向量 只能传标量
#所有的太极传参都是内存拷贝副本进行运算  外部数据不受印象
@ti.kernel
def my_kernel(x:ti.i32,y:ti.f32):
    print(x+y)

my_kernel(2,3.5)

#只能返回标量整形 会自动转为整形
@ti.kernel
def foo() -> ti.i32:
    return 233.4

print(foo())

